tensor([[0.2400, 0.2400, 0.2400, 0.2400, 0.2400],
        [0.7600, 0.7600, 0.7600, 0.7600, 0.7600]], grad_fn=<RoundBackward1>)